package club.xiaojiawei.graph;

import java.util.Arrays;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 6/14/22 7:54 PM
 * @question 684. 冗余连接
 * @description 树可以看成是一个连通且 无环 的 无向 图。
 * 给定往一棵 n 个节点 (节点值 1～n) 的树中添加一条边后的图。添加的边的两个顶点包含在 1 到 n 中间，且这条附加的边不属于树中已存在的边。图的信息记录于长度为 n 的二维数组 edges ，edges[i] = [ai, bi] 表示图中在 ai 和 bi 之间存在一条边。
 * 请找出一条可以删去的边，删除后可使得剩余部分是一个有着 n 个节点的树。如果有多个答案，则返回数组 edges 中最后出现的边。
 */
public class FindRedundantConnection684 {

    public static void main(String[] args) {
        FindRedundantConnection684 test = new FindRedundantConnection684();
        int[] result = test.findRedundantConnection(new int[][]{{1, 4},{1, 2}, {2, 3}, {3, 4},  {1, 5}});
        System.out.println(Arrays.toString(result));
    }

    /**
     * 官方-查并集
     * 时间复杂度 O(nlogn)
     * 空间复杂度 O(n)
     * @param edges
     * @return
     */
    public int[] findRedundantConnection(int[][] edges) {
        int n = edges.length;
        int[] parent = new int[n + 1];
        for (int i = 1; i <= n; i++) {
            parent[i] = i;
        }
        for (int[] edge : edges) {
            int node1 = edge[0], node2 = edge[1];
            if (find(parent, node1) != find(parent, node2)) {
                union(parent, node1, node2);
            } else {
                return edge;
            }
        }
        return new int[0];
    }

    public void union(int[] parent, int index1, int index2) {
        parent[find(parent, index1)] = find(parent, index2);
    }

    public int find(int[] parent, int index) {
        while (parent[index] != index){
            index = parent[index];
        }
        return index;
    }
}
